#! /usr/bin/env python
#coding:utf-8

import smtplib
from email.mime.text import MIMEText
from email.header import Header


def sendmail(login={}, mail={}):
    '''
    @param login login['user'] login['passwd']
    @param mail mail['to_addr'] mail['subject'] mail['content'] mail['attach']
    '''
    from datetime import datetime
    from base64 import b64encode
    import smtplib, mimetypes
    from email.mime.text import MIMEText
    from email.mime.multipart import MIMEMultipart
    from email.mime.image import MIMEImage

    user_info = login['user'].split('@')
    mail_configure = {}
    mail_configure['mail_encoding'] = 'utf-8'
    mail_configure['mail_supplier'] = user_info[1]
    mail_configure['from_addr'] = login['user']
    mail_configure['server_host'] = 'smtp.%s' % mail_configure['mail_supplier']
    error = None

    print mail_configure

    try:
        email = MIMEMultipart()
        email['from'] = mail_configure['from_addr']
        email['to'] = ','.join(mail['to_addr'])
        email['subject'] = '=?%s?B?%s?=' % (mail_configure['mail_encoding'],b64encode(mail['subject']))
        email_content = MIMEText(mail['content'], _charset=mail_configure['mail_encoding'])
        email.attach(email_content)

        if 'attach' in mail:
            for i in mail['attach']:
                ctype, encoding = mimetypes.guess_type(i)
                if ctype is None or not encoding is None:
                    ctype = 'application/octet-stream'
                maintype, subtype = ctype.split('/', 1)
                att = MIMEImage((lambda f: (f.read(), f.close()))(open(i, 'rb'))[0], _subtype = subtype)
                att.add_header('Content-Disposition', 'attachment', filename = i)
                email.attach(att)

        smtp = smtplib.SMTP()
        smtp.connect(mail_configure['server_host'])
        smtp.login(login['user'], login['passwd'])
        smtp.sendmail(mail_configure['from_addr'], mail['to_addr'], email.as_string())
        smtp.quit()
    except Exception as e:
        error = e
        print "sendmail error: %s" % str(e)

    return (mail_configure['from_addr'], mail['to_addr'], error)


def tdm_email_send_test1():
    login = {
        'user':'qujianning@100tal.com',
        'passwd':'asdfqwer123$'
    }
    mail = {
        'to_addr':'redfox_qu@163.com;redfox.qu@qq.com',
        'subject':'不带附件的测试邮件',
        'content':'''
                sz002718,友邦吊顶
                sz002719,麦趣尔
                sz002722,金轮股份
                ''',
    }
    print sendmail(login, mail)
    return

def tdm_email_send_test():
    fromaddr = 'qujianning@100tal.com'
    toaddrs = ['yanlong@100tal.com', 'chenzhongyi@100tal.com']
    subject = 'tdm alarm test'
    username = 'qujianning@100tal.com'
    password = 'asdf$'

    msg = MIMEText('服务器内存超限 > 80%', 'text', 'utf-8')
    msg['Subject'] = Header(subject, 'utf-8')

    smtp = smtplib.SMTP()
    smtp.connect('smtp.100tal.com')
    smtp.login(username, password)
    smtp.sendmail(fromaddr, toaddrs, msg.as_string())
    smtp.quit()

if __name__ == '__main__':
    tdm_email_send_test1()
